package com.bang.study.day02.test;

import java.util.ArrayList;
import java.util.HashMap;

/**
 * @Auther: Bang
 * @Date: 2019/9/14 21:38
 * @Description:  两个数组的交集 II
 *          给定两个数组，编写一个函数来计算它们的交集。
 *
             * 示例 1:
             *
             * 输入: nums1 = [1,2,2,1], nums2 = [2,2]
             * 输出: [2,2]
             * 示例 2:
             *
             * 输入: nums1 = [4,9,5], nums2 = [9,4,9,8,4]
             * 输出: [4,9]
             * 说明：
             *
             * 输出结果中每个元素出现的次数，应与元素在两个数组中出现的次数一致。
             * 我们可以不考虑输出结果的顺序。
 */
public class Solution350 {

    public int[] intersect(int[] nums1, int[] nums2) {
        HashMap<Integer, Integer> hashMap = new HashMap<>();
        for (int i = 0; i < nums1.length; i++) {
            int i1 = nums1[i];
            if (hashMap.containsKey(i1)) {
                hashMap.put(i1, hashMap.get(i1) + 1);
            }else{
                hashMap.put(i1, 1);
            }
        }
        ArrayList<Integer> list = new ArrayList<>();
        for (int i : nums2) {
            if (hashMap.containsKey(i)) {
                list.add(i);
                hashMap.put(i, hashMap.get(i) - 1);
                if ( hashMap.get(i)==0) {
                    hashMap.remove(i);
                }
            }
        }
        int size = list.size();
        int[] arr = new int[size];
        for (int i = 0; i < size; i++) {
            arr[i]=list.get(i);
        }
        return arr;
    }
}
